home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Ubuntu 9.10 PL / karmelkowy-koliberek-desktop-9.10-i386-PL.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / DistUpgradeViewText.py < prev    next >
Encoding:
Python Source  |  2009-11-02  |  10.0 KB  |  275 lines
  1. # DistUpgradeViewText.py 
  2. #  
  3. #  Copyright (c) 2004-2006 Canonical
  4. #  
  5. #  Author: Michael Vogt <michael.vogt@ubuntu.com>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. import sys
  23. import logging
  24. import subprocess
  25.  
  26. import apt
  27. import os
  28.  
  29. from DistUpgradeView import DistUpgradeView, InstallProgress, FetchProgress
  30. import apt.progress
  31.  
  32. import gettext
  33. from DistUpgradeGettext import gettext as _
  34. #from textwrap import fill, wrap
  35.  
  36. # helpers inspired after textwrap - unfortunately
  37. # we can not use textwrap directly because it break
  38. # packagenames with "-" in them into new lines
  39. def wrap(t, width=70, subsequent_indent=""):
  40.     out = ""
  41.     for s in t.split():
  42.         if (len(out)-out.rfind("\n")) + len(s) > width:
  43.             out += "\n" + subsequent_indent
  44.         out += s + " "
  45.     return out
  46.     
  47. def twrap(s, **kwargs):
  48.     msg = ""
  49.     paras = s.split("\n")
  50.     for par in paras:
  51.         s = wrap(par, **kwargs)
  52.         msg += s+"\n"
  53.     return msg
  54.  
  55. class TextFetchProgress(FetchProgress, apt.progress.TextFetchProgress):
  56.     def __init__(self):
  57.         apt.progress.TextFetchProgress.__init__(self)
  58.         FetchProgress.__init__(self)
  59.     def pulse(self):
  60.         apt.progress.TextFetchProgress.pulse(self)
  61.         FetchProgress.pulse(self)
  62.         return True
  63.  
  64. class TextCdromProgressAdapter(apt.progress.CdromProgress):
  65.     """ Report the cdrom add progress  """
  66.     def update(self, text, step):
  67.         """ update is called regularly so that the gui can be redrawn """
  68.         if text:
  69.           print "%s (%f)" % (text, step/float(self.totalSteps)*100)
  70.     def askCdromName(self):
  71.         return (False, "")
  72.     def changeCdrom(self):
  73.         return False
  74.  
  75.  
  76. class DistUpgradeViewText(DistUpgradeView):
  77.     " text frontend of the distUpgrade tool "
  78.     def __init__(self, datadir=None, logdir=None):
  79.         # its important to have a debconf frontend for
  80.         # packages like "quagga"
  81.         if not os.environ.has_key("DEBIAN_FRONTEND"):
  82.             os.environ["DEBIAN_FRONTEND"] = "dialog"
  83.         if not datadir:
  84.           localedir=os.path.join(os.getcwd(),"mo")
  85.         else:
  86.           localedir="/usr/share/locale/update-manager"
  87.  
  88.         try:
  89.           gettext.bindtextdomain("update-manager", localedir)
  90.           gettext.textdomain("update-manager")
  91.         except Exception, e:
  92.           logging.warning("Error setting locales (%s)" % e)
  93.         
  94.         self.last_step = 0 # keep a record of the latest step
  95.         self._opCacheProgress = apt.progress.OpTextProgress()
  96.         self._fetchProgress = TextFetchProgress()
  97.         self._cdromProgress = TextCdromProgressAdapter()
  98.         self._installProgress = InstallProgress()
  99.         sys.excepthook = self._handleException
  100.         #self._process_events_tick = 0
  101.  
  102.     def _handleException(self, type, value, tb):
  103.       import traceback
  104.       print
  105.       lines = traceback.format_exception(type, value, tb)
  106.       logging.error("not handled exception:\n%s" % "\n".join(lines))
  107.       self.error(_("A fatal error occurred"),
  108.                  _("Please report this as a bug and include the "
  109.                    "files /var/log/dist-upgrade/main.log and "
  110.                    "/var/log/dist-upgrade/apt.log "
  111.                    "in your report. The upgrade is now aborted.\n"
  112.                    "Your original sources.list was saved in "
  113.                    "/etc/apt/sources.list.distUpgrade."),
  114.                  "\n".join(lines))
  115.       sys.exit(1)
  116.  
  117.     def getFetchProgress(self):
  118.         return self._fetchProgress
  119.     def getInstallProgress(self, cache):
  120.         self._installProgress._cache = cache
  121.         return self._installProgress
  122.     def getOpCacheProgress(self):
  123.         return self._opCacheProgress
  124.     def getCdromProgress(self):
  125.         return self._cdromProgress
  126.     def updateStatus(self, msg):
  127.       print
  128.       print msg
  129.       sys.stdout.flush()
  130.     def abort(self):
  131.       print
  132.       print _("Aborting")
  133.     def setStep(self, step):
  134.       self.last_step = step
  135.     def showDemotions(self, summary, msg, demotions):
  136.         self.information(summary, msg, 
  137.                          _("Demoted:\n")+twrap(", ".join(demotions)))
  138.     def information(self, summary, msg, extended_msg=None):
  139.       print
  140.       print twrap(summary)
  141.       print twrap(msg)
  142.       if extended_msg:
  143.         print twrap(extended_msg)
  144.     def error(self, summary, msg, extended_msg=None):
  145.       print
  146.       print twrap(summary)
  147.       print twrap(msg)
  148.       if extended_msg:
  149.         print twrap(extended_msg)
  150.       return False
  151.     def showInPager(self, output):
  152.       " helper to show output in a pager"
  153.       for pager in ["/usr/bin/sensible-pager", "/bin/more"]:
  154.           if os.path.exists(pager):
  155.               p = subprocess.Popen([pager,"-"],stdin=subprocess.PIPE)
  156.               p.stdin.write(output)
  157.               p.stdin.close()
  158.               p.wait()
  159.               return
  160.       # if we don't have a pager, just print
  161.       print output
  162.  
  163.     def confirmChanges(self, summary, changes, downloadSize,
  164.                        actions=None, removal_bold=True):
  165.       DistUpgradeView.confirmChanges(self, summary, changes, downloadSize, actions)
  166.       print
  167.       print twrap(summary)
  168.       print twrap(self.confirmChangesMessage)
  169.       print " %s %s" % (_("Continue [yN] "), _("Details [d]")),
  170.       while True:
  171.         res = sys.stdin.readline()
  172.         # TRANSLATORS: the "y" is "yes"
  173.         if res.strip().lower().startswith(_("y")):
  174.           return True
  175.         # TRANSLATORS: the "n" is "no"
  176.         elif res.strip().lower().startswith(_("n")):
  177.           return False
  178.         # TRANSLATORS: the "d" is "details"
  179.         elif res.strip().lower().startswith(_("d")):
  180.           output = ""
  181.           if len(self.toRemove) > 0:
  182.               output += "\n"  
  183.               output += twrap(_("Remove: %s\n" % " ".join(self.toRemove)), subsequent_indent='  ')
  184.           if len(self.toInstall) > 0:
  185.               output += "\n"
  186.               output += twrap(_("Install: %s\n" % " ".join(self.toInstall)), subsequent_indent='  ')
  187.           if len(self.toUpgrade) > 0:
  188.               output += "\n"  
  189.               output += twrap(_("Upgrade: %s\n" % " ".join(self.toUpgrade)), subsequent_indent='  ')
  190.           self.showInPager(output)
  191.         print "%s %s" % (_("Continue [yN] "), _("Details [d]")),
  192.  
  193.     def askYesNoQuestion(self, summary, msg, default='No'):
  194.       print
  195.       print twrap(summary)
  196.       print twrap(msg)
  197.       if default == 'No':
  198.           print _("Continue [yN] "),
  199.           res = sys.stdin.readline()
  200.           # TRANSLATORS: first letter of a positive (yes) answer
  201.           if res.strip().lower().startswith(_("y")):
  202.               return True
  203.           return False
  204.       else:
  205.           print _("Continue [Yn] "),
  206.           res = sys.stdin.readline()
  207.           # TRANSLATORS: first letter of a negative (no) answer
  208.           if res.strip().lower().startswith(_("n")):
  209.               return False
  210.           return True
  211.  
  212. # FIXME: when we need this most the resolver is writing debug logs
  213. #        and we redirect stdout/stderr    
  214. #    def processEvents(self):
  215. #      #time.sleep(0.2)
  216. #      anim = [".","o","O","o"]
  217. #      anim = ["\\","|","/","-","\\","|","/","-"]
  218. #      self._process_events_tick += 1
  219. #      if self._process_events_tick >= len(anim):
  220. #          self._process_events_tick = 0
  221. #      sys.stdout.write("[%s]" % anim[self._process_events_tick])
  222. #      sys.stdout.flush()
  223.  
  224.     def confirmRestart(self):
  225.       return self.askYesNoQuestion(_("Restart required"),
  226.                                    _("To finish the upgrade, a restart is "
  227.                                      "required.\n"
  228.                                      "If you select 'y' the system "
  229.                                      "will be restarted."), default='No')
  230.  
  231.  
  232. if __name__ == "__main__":
  233.   view = DistUpgradeViewText()
  234.  
  235.   #while True:
  236.   #    view.processEvents()
  237.   
  238.   print twrap("89 packages are going to be upgraded.\nYou have to download a total of 82.7M.\nThis download will take about 10 minutes with a 1Mbit DSL connection and about 3 hours 12 minutes with a 56k modem.", subsequent_indent=" ")
  239.   #sys.exit(1)
  240.  
  241.   view = DistUpgradeViewText()
  242.   print view.askYesNoQuestion("hello", "Icecream?", "No")
  243.   print view.askYesNoQuestion("hello", "Icecream?", "Yes")
  244.   
  245.  
  246.   #view.confirmChangesMessage = "89 packages are going to be upgraded.\n You have to download a total of 82.7M.\n This download will take about 10 minutes with a 1Mbit DSL connection and about 3 hours 12 minutes with a 56k modem."
  247.   #view.confirmChanges("xx",[], 100)
  248.   sys.exit(0)
  249.  
  250.   view.confirmRestart()
  251.  
  252.   cache = apt.Cache()
  253.   fp = view.getFetchProgress()
  254.   ip = view.getInstallProgress(cache)
  255.  
  256.  
  257.   for pkg in sys.argv[1:]:
  258.     cache[pkg].markInstall()
  259.   cache.commit(fp,ip)
  260.   
  261.   sys.exit(0)
  262.   view.getTerminal().call(["dpkg","--configure","-a"])
  263.   #view.getTerminal().call(["ls","-R","/usr"])
  264.   view.error("short","long",
  265.              "asfds afsdj af asdf asdf asf dsa fadsf asdf as fasf sextended\n"
  266.              "asfds afsdj af asdf asdf asf dsa fadsf asdf as fasf sextended\n"
  267.              "asfds afsdj af asdf asdf asf dsa fadsf asdf as fasf sextended\n"
  268.              "asfds afsdj af asdf asdf asf dsa fadsf asdf as fasf sextended\n"
  269.              "asfds afsdj af asdf asdf asf dsa fadsf asdf as fasf sextended\n"
  270.              "asfds afsdj af asdf asdf asf dsa fadsf asdf as fasf sextended\n"
  271.              "asfds afsdj af asdf asdf asf dsa fadsf asdf as fasf sextended\n"
  272.              )
  273.   view.confirmChanges("xx",[], 100)
  274.   print view.askYesNoQuestion("hello", "Icecream?")
  275.